import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt
mpl.use('TkAgg')

# 建立画布
# fig, (ax1, ax2, ax3) = plt.subplots(1, 3, figsize=(15, 5))

# 数据读取
axial_path = r'../data/MRNet-v1.0/train/axial/0000.npy'
coronal_path = r'../data/MRNet-v1.0/train/coronal/0000.npy'
sagittal_path = r'../data/MRNet-v1.0/train/sagittal/0000.npy'
axial_mri = np.load(axial_path)

coronal_mri = np.load(coronal_path)
sagittal_mri = np.load(sagittal_path)

plt.imshow(sagittal_mri[2, :, :], 'gray')

# i = 20
# ax1.imshow(axial_mri[i, :, :], 'gray')
# ax1.set_title('Case 0 | Slice 1 | Axial')
#
# ax2.imshow(coronal_mri[i, :, :], 'gray')
# ax2.set_title('Case 0 | Slice 1 | Coronal')
#
# ax3.imshow(sagittal_mri[i, :, :], 'gray')
# ax3.set_title('Case 0 | Slice 1 | Sagittal')

plt.show()
